package com.swt.agy.backend.utils;

import org.apache.commons.lang.NullArgumentException;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;


/**
 * FTP客户端工具类
 * @author luolin
 *
 * @version $id:FTPClientUtils.java,v 0.1 2015年11月13日 下午4:18:07 luolin Exp $
 */
public class FTPClientUtil {

    public String addr = "";
    public int port = 21;
    public String username = "ag";
    public String password = "1234567";

    /**
     * 连接文件服务器
     * @param addr 文件服务器地址
     * @param port 端口
     * @param username 用户名
     * @param password 密码
     * @throws Exception
     */
    public static FTPClient connect(String addr, int port, String username, String password) {


        FTPClient ftpClient = new FTPClient();
        try {
            // 连接
            ftpClient.connect(addr, port);
            // 登录
            ftpClient.login(username, password);
            // 被动模式：每次数据连接之前，ftp client告诉ftp server开通一个端口来传输数据（参考资料：FTP主动/被动模式的解释）
            ftpClient.enterLocalPassiveMode();
            ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
        } catch (Exception e) {

            throw new RuntimeException("连接文件服务器失败");
        }
        // 判断文件服务器是否可用？？
        if (!FTPReply.isPositiveCompletion(ftpClient.getReplyCode())) {
            closeConnection(ftpClient);
        }
        return ftpClient;
    }

    /**
     * 连接文件服务器
     * @param addr 文件服务器地址
     * @param port 端口
     * @param username 用户名
     * @param password 密码
     * @param workingDirectory 目标连接工作目录
     * @throws Exception
     */
    public static FTPClient connect(String addr, int port, String username, String password, String workingDirectory)
            throws Exception {
        FTPClient ftpClient = connect(addr, port, username, password);
        changeWorkingDirectory(workingDirectory, ftpClient);
        return ftpClient;
    }

    /**
     * 关闭连接，使用完连接之后，一定要关闭连接，否则服务器会抛出 Connection reset by peer的错误
     * @throws IOException
     */
    public static void closeConnection(FTPClient ftpClient) {

        if (ftpClient == null) {
            return;
        }

        try {
            ftpClient.disconnect();
        } catch (IOException e) {

            throw new RuntimeException("关闭连接失败");
        }
    }

    /**
     * 切换工作目录
     * @param directory 目标工作目录
     * @param ftpClient
     * @throws IOException
     */
    public static void changeWorkingDirectory(String directory, FTPClient ftpClient) {

        baseValidate(ftpClient);
        // 切换到目标工作目录
        try {
            if (!ftpClient.changeWorkingDirectory(directory)) {
                ftpClient.makeDirectory(directory);
                ftpClient.changeWorkingDirectory(directory);
            }
        } catch (Throwable e) {

            throw new RuntimeException("切换工作目录失败");
        }
    }

    /**
     * 上传文件/文件夹
     * @param file 上传的文件或文件夹
     * @return 文件存放的路径以及文件名
     * @throws Exception
     */
    public static void upload(File file, FTPClient ftpClient) throws Exception {
        if (file == null) {

            throw new RuntimeException("上传文件为空");
        }

        baseValidate(ftpClient);
        // 是文件，直接上传
        if (!file.isDirectory()) {
            storeFile(new File(file.getPath()), ftpClient);
            return;
        }

        changeWorkingDirectory(file.getName(), ftpClient);
        // 文件夹，递归上传所有文件
        for (File item : file.listFiles()) {
            if (!item.isDirectory()) {
                storeFile(item, ftpClient);
                continue;
            }
            upload(item, ftpClient);
            ftpClient.changeToParentDirectory();
        }
    }

    /**
     * 删除文件
     * @param fileName 要删除的文件地址
     * @return true/false
     * @throws IOException
     */
    public static boolean delete(String fileName, FTPClient ftpClient) throws IOException {

        baseValidate(ftpClient);
        if (StringUtils.isBlank(fileName)) {

            throw new NullArgumentException("fileName");
        }
        return ftpClient.deleteFile(fileName);
    }

    /**
     * 存储文件
     * @param file {@link File}
     * @throws Exception
     */
    public static void storeFile(File file, FTPClient ftpClient) throws Exception {
        if (file == null) {

            throw new RuntimeException("存储的文件为空");
        }

        baseValidate(ftpClient);

        FileInputStream input = new FileInputStream(file);
        ftpClient.storeFile(file.getName(), input);
        input.close();
    }

    /**
     * 存储文件
     * @param inputStream {@link InputStream}
     * @param fileName 文件名
     * @throws Exception
     */
    public static void storeFile(InputStream inputStream, String fileName, FTPClient ftpClient) throws Exception {

        if (inputStream == null) {

            throw new RuntimeException("存储的文件为空");
        }
        baseValidate(ftpClient);
        ftpClient.storeFile(fileName, inputStream);
        inputStream.close();
    }

    /**
     * 下载文件到指定目录
     * @param ftpFile 文件服务器上的文件地址
     * @param dstFile 输出文件的路径和名称
     * @throws Exception
     */
    public static void downLoad(String ftpFile, String dstFile, FTPClient ftpClient) throws Exception {

        if (StringUtils.isBlank(ftpFile)) {

            throw new RuntimeException("【参数ftpFile为空】");
        }
        if (StringUtils.isBlank(dstFile)) {

            throw new RuntimeException("【参数dstFile为空】");
        }
        baseValidate(ftpClient);
        File file = new File(dstFile);
        FileOutputStream fos = new FileOutputStream(file);
        ftpClient.retrieveFile(ftpFile, fos);
        fos.flush();
        fos.close();
    }

    /**
     * 从文件服务器获取文件流
     * @param ftpFile 文件服务器上的文件地址
     * @return {@link InputStream}
     * @throws IOException
     */
    public static InputStream retrieveFileStream(String ftpFile, FTPClient ftpClient) throws IOException {

        if (StringUtils.isBlank(ftpFile)) {

            throw new RuntimeException("【参数ftpFile为空】");
        }
        baseValidate(ftpClient);
        return ftpClient.retrieveFileStream(ftpFile);
    }


    /**
     * 复制文件
     * @param ftpClient {@link FTPClient}
     * @param sourceFile 源文件
     * @param targetDir 目标文件夹
     * @param newName 新的文件名
     * @return 复制后的文件路径及文件名
     * @throws Exception
     */
    public static String copy(FTPClient ftpClient, String sourceFile, String targetDir, String newName)
            throws Exception {
        if (StringUtils.isBlank(sourceFile)) {

            throw new NullArgumentException("sourceFile");
        }
        if (StringUtils.isBlank(targetDir)) {

            throw new NullArgumentException("targetDir");
        }
        if (StringUtils.isBlank(newName)) {

            throw new NullArgumentException("newName");
        }
        baseValidate(ftpClient);

        InputStream is = ftpClient.retrieveFileStream(sourceFile);
        // 主动调用一次getReply()把接下来的226消费掉. 这样做是可以解决这个返回null问题
        ftpClient.getReply();
        if (is == null) {

            throw new RuntimeException("未找到源文件");
        }

        changeWorkingDirectory(targetDir, ftpClient);

        storeFile(is, newName, ftpClient);

        return ftpClient.printWorkingDirectory() + File.separator + newName;
    }

    /**
     * 基本校验
     * @param ftpClient {@link FTPClient}
     */
    private static void baseValidate(FTPClient ftpClient) {
        if (ftpClient == null) {

            throw new NullArgumentException("ftpClient");
        }
    }
}
